import requests
import os
import sys
import time

from contextlib import closing
from util.log_handler import LogHandler

from util.log_handler import LogHandler
from util.util_function import DownloadFile

# 获取文件名
def get_filename(path):
	filename = os.path.basename(path)
	filename = filename.split('.')[0]
	return filename

# 获取当前执行程序的文件名
def get_execute_filename():
	return get_filename(__file__)

log = LogHandler(get_execute_filename())

def file_readlines(file_path, mode='r', encoding='utf8'):
    '''
    1. 作用:逐行读取文件内容，并保存到列表中，列表的一个元素为文件一行的内容
    2. 参数：
        file_path:文件路径
        mode:文件读方式,默认为'r'
        encoding:文件编码,默认为'utf8'
    3. 返回值:逐行读取文件内容的列表
    '''
    f = open(file_path, mode=mode, encoding=encoding)
    file_content = f.readlines()
    f.close()
    return file_content

if __name__ == '__main__':
    list_img_info = file_readlines('/home/ubuntu/workspace/guatian/liuliang.csv')
    dir_name = '/home/ubuntu/workspace/guatian/guatian_img'
    print(list_img_info)

    for img_info in list_img_info:
        img_name = img_info.split(',')[0]
        img_url = img_info.split(',')[1]
        print(img_name, img_url)
        DownloadFile(img_url=img_url, dir_name=dir_name, img_name=img_name)
